'''fcn_resnest101os8_inria'''
import os
from .base_cfg import *


# modify dataset config
DATASET_CFG = DATASET_CFG.copy()
DATASET_CFG.update({
    'type': 'inria',
    'rootdir': '/data/SemanticSegmnetation/Inria',
})
# modify dataloader config
DATALOADER_CFG = DATALOADER_CFG.copy()
DATALOADER_CFG['train'].update({
    'batch_size': 4,
})
# modify optimizer config
OPTIMIZER_CFG = OPTIMIZER_CFG.copy()
# modify scheduler config
SCHEDULER_CFG = SCHEDULER_CFG.copy()
SCHEDULER_CFG.update({
    'max_epochs': 130
})
# modify losses config
LOSSES_CFG = LOSSES_CFG.copy()
# modify segmentor config
SEGMENTOR_CFG = SEGMENTOR_CFG.copy()
SEGMENTOR_CFG.update({
    'num_classes': 2,
    'backbone': {
        'type': 'resnest101',
        'series': 'resnest',
        'pretrained': False,
        'outstride': 8,
        'selected_indices': (2, 3),
    },
    'auxiliary': {
        # 'in_channels': 1024,
        'in_channels': 1024,
        'out_channels': 512,
        'dropout': 0.1,
    }
})
# modify inference config
INFERENCE_CFG = INFERENCE_CFG.copy()
# modify common config
COMMON_CFG = COMMON_CFG.copy()
COMMON_CFG['work_dir'] = 'fcn_resnest101os8_inria'
COMMON_CFG['logfilepath'] = 'fcn_resnest101os8_inria/fcn_resnest101os8_inria.log'
COMMON_CFG['resultsavepath'] = 'fcn_resnest101os8_inria/fcn_resnest101os8_inria_results.pkl'